package grpcclient

import (
	"context"
	"reflect"
	"strings"

	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

var _ grpc.ClientConnInterface = &localConn{}

// NewLocalGRPCClient 创建一个直接调用服务器的gRPC客户端
// !!! 警告: 它不进行任何网络传输或序列化/反序列化操作，因此不保证并发安全
// 并且与普通的gRPC客户端/服务器不同，这里没有客户端和服务器的拦截器
func NewLocalGRPCClient[C any, S any](desc *grpc.ServiceDesc, server S, clientCreator func(grpc.ClientConnInterface) C) C {
	return clientCreator(&localConn{
		serviceDesc: desc,
		server:      server,
	})
}

// localConn 是 grpc.ClientConnInterface 的实现，它直接调用服务器
type localConn struct {
	serviceDesc *grpc.ServiceDesc // ServiceDesc 是此服务的描述符
	server      interface{}       // 服务器对象
}

// Invoke 直接调用服务器方法
func (c *localConn) Invoke(ctx context.Context, method string, args, reply interface{}, opts ...grpc.CallOption) error {
	methodDesc := c.findMethod(method)
	if methodDesc == nil {
		return status.Errorf(codes.Unimplemented, "method %s not implemented", method)
	}
	// 调用找到的gRPC方法的处理函数
	resp, err := methodDesc.Handler(c.server, ctx, func(in any) error {
		reflect.ValueOf(in).Elem().Set(reflect.ValueOf(args).Elem())
		return nil
	}, nil)
	if err != nil {
		return err
	}
	//  将resp的值赋给reply指向的内存位置
	reflect.ValueOf(reply).Elem().Set(reflect.ValueOf(resp).Elem())
	return nil
}

// NewStream 目前不支持，等待实现
func (c *localConn) NewStream(ctx context.Context, desc *grpc.StreamDesc, method string, opts ...grpc.CallOption) (grpc.ClientStream, error) {
	panic("we don't support local stream rpc by now")
}

// findMethod 根据完整的方法名查找对应的方法描述符
// fullMethodName: 完整的方法名，格式为 "/serviceName/methodName"
// 返回值: 对应的gRPC方法描述符，如果找不到则返回nil
func (c *localConn) findMethod(fullMethodName string) *grpc.MethodDesc {
	// 去掉开头的'/'并按'/'分割成服务名和方法名两部分
	strs := strings.SplitN(fullMethodName[1:], "/", 2)
	serviceName := strs[0]  // 获取服务名
	methodName := strs[1]   // 获取方法名
	
	// 检查服务名是否匹配当前连接的服务描述符
	if c.serviceDesc.ServiceName != serviceName {
		return nil
	}
	
	// 遍历服务描述符中的所有方法，查找匹配的方法名
	for i := range c.serviceDesc.Methods {
		if c.serviceDesc.Methods[i].MethodName == methodName {
			return &c.serviceDesc.Methods[i]  // 找到匹配的方法，返回其描述符
		}
	}
	return nil  // 未找到匹配的方法
}